from pylab import *


xList = [1,2,3,4,5] 
yList = [2,3,4,5,6]

pfad1 = "output_gw_on.txt" 
pfad2 = "output_gw_off.txt"

in_file1 = open(pfad1,"r")
in_file2 = open(pfad2,"r")

text1 = in_file1.read()
text2 = in_file2.read()

in_file1.close()
in_file2.close()

verbrauchList1 = text1.splitlines()
verbrauchList2 = text2.splitlines()

benzinverbrauch21 = []
zuWeg1 = []
counter1 = 0
autoList1 = []
gesamtZeit1 = []
warteZeit1 = []
gesamtZeitAlle1 = 0
warteZeitAlle1 = 0
intWeg = 0
intBenzin = 0
intZeit = 0
intWartezeit = 0
intCo2 = 0

# fuer Alg. 1

for e in verbrauchList1:
    b = e[0:22]
    if (b == "Durchschnittsverbrauch"):
       benzinverbrauch = e.rsplit(' ')
       intBenzin+=float(benzinverbrauch[1])
       benzinverbrauch21.append(benzinverbrauch[1])
       counter1 += 1
       autoList1.append(counter1)
    b = e[0:10]   
    if (b == "Gesamtzeit"):
       gesamtzeit = e.rsplit(' ')
       intZeit+= float(gesamtzeit[1])
       gesamtZeit1.append(gesamtzeit[1])
       gesamtZeitAlle1 += float(gesamtzeit[1])
    b = e[0:9]
    if (b == "Wartezeit"):
        wartezeit = e.rsplit(' ')
        intWartezeit += float(wartezeit[1])
        warteZeit1.append(wartezeit[1])
        warteZeitAlle1 += float(wartezeit[1])

gesamtZeitAlle1 = gesamtZeitAlle1 - warteZeitAlle1
       

summe1 = 0
summeList1 = []
for a in benzinverbrauch21:
    summe1 += float(a)
summe1 = summe1/len(benzinverbrauch21)
for x in range(len(benzinverbrauch21)):
    summeList1.append(summe1)           
print intBenzin
print intZeit
print intWartezeit

# fuer Alg. 2

benzinverbrauch22 = []
zuWeg2 = []
counter2 = 0
autoList2 = []
gesamtZeit2 = []
warteZeit2 = []
gesamtZeitAlle2 = 0
warteZeitAlle2 = 0
intBenzin = 0
intZeit = 0
intWartezeit = 0


for e in verbrauchList2:
    b = e[0:22]
    if (b == "Durchschnittsverbrauch"):
       benzinverbrauch = e.rsplit(' ')
       intBenzin+=float(benzinverbrauch[1])
       benzinverbrauch22.append(benzinverbrauch[1])
       counter2 += 1
       autoList2.append(counter2)
    b = e[0:10]   
    if (b == "Gesamtzeit"):
       gesamtzeit = e.rsplit(' ')
       intZeit+= float(gesamtzeit[1])
       gesamtZeit2.append(gesamtzeit[1])
       gesamtZeitAlle2 += float(gesamtzeit[1])
    b = e[0:9]
    if (b == "Wartezeit"):
        wartezeit = e.rsplit(' ')
        intWartezeit += float(wartezeit[1])
        warteZeit2.append(wartezeit[1])
        warteZeitAlle2 += float(wartezeit[1])

gesamtZeitAlle2 = gesamtZeitAlle2 - warteZeitAlle2
       

summe2 = 0
summeList2 = []
for a in benzinverbrauch22:
    summe2 += float(a)
summe2 = summe2/len(benzinverbrauch22)
for x in range(len(benzinverbrauch22)):
    summeList2.append(summe2) 
    
print intBenzin
print intZeit
print intWartezeit

# Zeichnen
# Benzinverbrauch 
subplot(321)
plot(autoList1, benzinverbrauch21, label="Benzinverbrauch Alg. 1")
ylabel('Verbrauch in Liter')
xlabel('Autoanzahl')
title('Benzinverbrauch')

subplot(321)
plot(autoList2, benzinverbrauch22, label="Benzinverbrauch Alg. 2")

subplot(321)
plot(autoList1, summeList1, label="Durchschnittsverbrauch Alg. 1")

subplot(321)
plot(autoList2, summeList2, label="Durchschnittsverbrauch Alg. 2")
legend(bbox_to_anchor = (0.8, -0.2))

# Wartezeit
subplot(322)
plot(autoList1, warteZeit1, label="Wartezeit Alg. 1")
ylabel('Zeit')
xlabel('Autoanzahl')
title('Gesamtzeit / Wartezeit')

subplot(322)
plot(autoList2, warteZeit2, label="Wartezeit Alg. 2", )

subplot(322)
plot(autoList1, gesamtZeit1, label ="Gesamtzeit Alg. 1")

subplot(322)
plot(autoList2, gesamtZeit2, label ="Gesamtzeit Alg. 2")
#legend(bbox_to_anchor = (0.7, -0.2))
#legend(loc=0, prop = FontProperties(size='smaller'))
legend (loc=0)





subplot(325)
pie([gesamtZeitAlle1, warteZeitAlle1], explode=None, labels=["Fahrzeit", "Wartezeit"],
    colors=('b', 'r', 'c', 'm', 'y', 'k', 'w'),
    autopct=None, pctdistance=0.6, labeldistance=1.1, shadow=False)
title("Gesamtzeit Alg. 1")

subplot(326)
pie([gesamtZeitAlle2, warteZeitAlle2], explode=None, labels=["Fahrzeit", "Wartezeit"],
    colors=('b', 'r', 'c', 'm', 'y', 'k', 'w'),
    autopct=None, pctdistance=0.6, labeldistance=1.1, shadow=False)
title("Gesamtzeit Alg. 2")



#subplot(327)
#bar(10, 10, width=0.8, bottom=0,
#    color=None, edgecolor=None, linewidth=None,
#    yerr=None, xerr=None, ecolor=None, capsize=3,
#    align='edge', orientation='vertical', log=False)
#title("Bla")



show()


